import requests
import json
domain = "comm.chatglm.cn"

def get_completion(prompt):
    headers = {
         'Content-Type': 'application/json',
         'Authorization' : 'Bearer DE32EB46D23350E0B75B3702C8D5DC036F8CE32DC73C919D'
    }
    data = {"company_name": prompt}
    response = requests.post(url=f'https://{domain}/law_api/get_company_register', headers=headers, json=data)
    return response.json()

if __name__ == '__main__':
 print(get_completion('广州发展集团股份有限公司'))